import request from '@/config/axios'

// 数据集类型 VO
export interface DatasetTypeVO {
  id?: number | undefined // promptID
  name: string | undefined // 数据集类型名称
  parentId: number | undefined // 父数据集类型id
  level: number // 等级 1 2 3 4 5 6...
}

// 数据集类型 API
export const DatasetTypeApi = {
  // 查询数据集类型分页
  getDatasetTypeList: async () => {
    return await request.get({ url: `/multiModel/dataset-type/list` })
  },

  // 查询数据集类型(一二三级)列表 -- 首页
  getDatasetTypeTop3: async () => {
    return await request.get({ url: `/multiModel/dataset-type/level-list` })
  },

  // 查询数据集类型详情
  getDatasetType: async (id: number) => {
    return await request.get({ url: `/multiModel/dataset-type/get?id=` + id })
  },

  // 新增数据集类型
  createDatasetType: async (data: DatasetTypeVO) => {
    return await request.post({ url: `/multiModel/dataset-type/create`, data })
  },

  // 修改数据集类型
  updateDatasetType: async (data: DatasetTypeVO) => {
    return await request.put({ url: `/multiModel/dataset-type/update`, data })
  },

  // 删除数据集类型
  deleteDatasetType: async (id: number) => {
    return await request.delete({ url: `/multiModel/dataset-type/delete?id=` + id })
  },

  // 导出数据集类型 Excel
  exportDatasetType: async (params) => {
    return await request.download({ url: `/multiModel/dataset-type/export-excel`, params })
  }
}
